# 配置文件
import os
from dotenv import load_dotenv

# 加载环境变量
load_dotenv()

# 星河大模型 API 配置
XINGHE_API_KEY = os.getenv('XINGHE_API_KEY', '')
XINGHE_BASE_URL = os.getenv('XINGHE_BASE_URL', '')

# 模型配置
XINGHE_MODEL = "ernie-4.5-turbo-vl"  # 星河大模型名称

# 应用配置
FLASK_SECRET_KEY = 'your-secret-key-here'
DEBUG_MODE = True

# 默认用户配置
DEFAULT_USERS = {
    'admin': {'password': '123456', 'role': 'admin', 'name': '管理员'},
    'student': {'password': '123456', 'role': 'student', 'name': '学生'},
    'teacher': {'password': '123456', 'role': 'teacher', 'name': '教师'}
}

# 知识库配置
KNOWLEDGE_BASE_PATH = 'rust_knowledge_base/rust_docs_sample.json'

# 问题分类标签
QUESTION_LABELS = ["greeting", "definition", "usage", "error_debug", "comparison", "faq"]

# 从环境变量读取API密钥（如果存在）
if os.getenv('XINGHE_API_KEY'):
    XINGHE_API_KEY = os.getenv('XINGHE_API_KEY')